#ifndef PAGEDATTENTIONBINDER_H
#define PAGEDATTENTIONBINDER_H 
#include "hosttensor_binder.h" 
#include <vector> 
namespace atb_speed { 
class PagedAttentionBinder : public HostTensorBinder { 
public: 
    PagedAttentionBinder(); 
    virtual ~PagedAttentionBinder(); 
    void ParseParam(const nlohmann::json &paramJson) override; 
    void BindTensor(atb::VariantPack &variantPack) override; 
    
private: 
    std::vector<int32_t> contextLens_; 
    std::vector<int32_t> batchRunStatus_; 
}; 
} 
#endif


